#!/usr/bin/env python
# -*-coding:UTF-8 -*-
'''
@Project ：爬虫-波波老师
@File：08_qiutubaike.py
@Author ：文非
@Date：2021/3/1017:09
'''
# 需求 爬取糗事百科中热图模块下的所有糗事图片
import requests
import re
import os  # 创建一个文件夹保存所有图片

if __name__== "__main__":

    # 创建一个文件夹，保存所有的爬取下来的图片
    if not os.path.exists("./糗事百科图片集"):  # os模块方法 如果这个文件夹不存在
        os.mkdir("./糗事百科图片集") # 在当前目录下新建一个文件夹

    # url
    url = "https://www.qiushibaike.com/imgrank/"
    # ua伪装
    headers = {
        'User-Agent': "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.100 Safari/537.36"
    }
    # 发送请求
    # 使用通用爬虫对url对应的一整张页面进行爬取
    pageText = requests.get(url=url, headers=headers).text
    # 获取响应数据# 持久保存数据
    # with open("./qiushi.html", "w", encoding="utf-8") as fp:
    #     fp.write(pageText)
    # 使用聚焦爬虫 对页面中的指定内容进行爬取 ---糗事百科每个段子的图片
    # 使用正则提取class = trumb的标签中的img标签中的链接
    ex = '<div class="thumb">.*?<img src="(.*?)" alt.*?</div>'
    # re.S 就是把\n也识别到 是多行匹配 re.M是单行匹配 在爬虫数据解析中使用的一定是re.S
    img_src_List = re.findall(ex, pageText, re.S)  # re.findall()方法返回的是一个列表、
    print(img_src_List)
    for src in img_src_List:
        # 拼接一个完整的图片地址url
        src = "https:" + src
        # 对这个图片地址发送get请求
        img_data = requests.get(url=src, headers=headers).content
        # 持久化存储
        # 生成图片名称
        img_name = src.split("/")[-1]
        # 图片存储的路径
        img_path = "./糗事百科图片集/" + img_name
        with open(img_path,"wb")as fp:
            fp.write(img_data)
            print(img_name,"下载成功")




